tensor([[0.4200, 0.4200, 0.4200, 0.4200, 0.4200],
        [0.5800, 0.5800, 0.5800, 0.5800, 0.5800]], grad_fn=<RoundBackward1>)